gpu=0
bs=32
dl_workers=4
src_path="data/ImageNet/ILSVRC2012_img_val"

# python scripts/metrics/prepare_reconstruction_img.py \
#     --device_num $gpu \
#     --src_path $src_path \
#     --bs $bs \
#     --dl_workers $dl_workers \
#     --cfg_path configs/custom/inference.yaml \
#     --ckpt_path "logs/VQModel/2023-02-09T09-15-54_OI_res256_n16384_d4/checkpoints/epoch=000000.ckpt" \
#     --save_dir samples/metric_samples/vq_custom_base

python scripts/metrics/prepare_reconstruction_img.py \
    --device_num $gpu \
    --src_path $src_path \
    --bs $bs \
    --dl_workers $dl_workers \
    --cfg_path configs/vq-f4/inference.yaml \
    --ckpt_path "logs/vq-f4/base/2023-02-22T10-46-21_base/checkpoints/best_s120000.ckpt" \
    --save_dir samples/metric_samples/vq_f4_base/res_256 \
    --resolution 256

python scripts/metrics/prepare_reconstruction_img.py \
    --device_num $gpu \
    --src_path $src_path \
    --bs $bs \
    --dl_workers $dl_workers \
    --cfg_path configs/vq-f4/inference.yaml \
    --ckpt_path "logs/vq-f4/base/2023-02-22T10-46-21_base/checkpoints/best_s120000.ckpt" \
    --save_dir samples/metric_samples/vq_f4_base/res_512 \
    --resolution 512

python scripts/metrics/prepare_reconstruction_img.py \
    --device_num $gpu \
    --src_path $src_path \
    --bs $bs \
    --dl_workers $dl_workers \
    --cfg_path configs/vq-f4-noattn/inference.yaml \
    --ckpt_path "logs/vq-f4-noattn/base/2023-02-22T10-48-11_base/checkpoints/best_s160000.ckpt" \
    --save_dir samples/metric_samples/vq_f4_noattn_base/res_256 \
    --resolution 256

python scripts/metrics/prepare_reconstruction_img.py \
    --device_num $gpu \
    --src_path $src_path \
    --bs $bs \
    --dl_workers $dl_workers \
    --cfg_path configs/vq-f4-noattn/inference.yaml \
    --ckpt_path "logs/vq-f4-noattn/base/2023-02-22T10-48-11_base/checkpoints/best_s160000.ckpt" \
    --save_dir samples/metric_samples/vq_f4_noattn_base/res_512 \
    --resolution 512

python scripts/metrics/prepare_reconstruction_img.py \
    --device_num $gpu \
    --src_path $src_path \
    --bs $bs \
    --dl_workers $dl_workers \
    --cfg_path configs/vq-f8/inference.yaml \
    --ckpt_path "logs/vq-f8/base/2023-02-27T19-46-36_base/checkpoints/best_s320000.ckpt" \
    --save_dir samples/metric_samples/vq_f8_base

python scripts/metrics/prepare_reconstruction_img.py \
    --device_num $gpu \
    --src_path $src_path \
    --bs $bs \
    --dl_workers $dl_workers \
    --cfg_path configs/vq-f8-n256/inference.yaml \
    --ckpt_path "logs/vq-f8-n256/base/2023-02-24T13-09-46_base/checkpoints/best_s240000.ckpt" \
    --save_dir samples/metric_samples/vq_f8_n256_base

# python scripts/metrics/prepare_reconstruction_img.py \
#     --device_num $gpu \
#     --src_path $src_path \
#     --bs $bs \
#     --dl_workers $dl_workers \
#     --cfg_path configs/vq-f16/inference.yaml \
#     --ckpt_path "logs/vq-f16/base/2023-02-19T13-05-25_base/checkpoints/last.ckpt" \
#     --save_dir samples/metric_samples/vq_f16_base